#!/bin/bash
#SBATCH -J task_wgan_param
#SBATCH -p gpu-normal
#SBATCH -N 1
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH -o task_wgan_param.out
#SBATCH -e task_wgan_param.err


module load anaconda3/2019.10 cuda/10.1

cd ..
pwd

python train.py --gpu_id=0 --output_dir=output_wgan_param --GAN_type=WGAN  --w_clip_c=0.03 --w_lr=0.0001 --w_d_iter=2

